package com.maikobi.maikobi.utils;

import jakarta.servlet.http.HttpServletRequest;

import java.net.InetAddress;

/**
 * 网络工具类
 *
 * @author: Maiko7
 * @create: 2025-07-16 7:57
 */
public class NetUtils {

    /**
     * 获取客户端 IP 地址
     * <p>
     * 该方法通过检查各种请求头来确定客户端的真实 IP 地址
     * 它考虑了代理服务器和负载平衡器可能存在的场景
     *
     * @param request HTTP 请求对象，用于获取请求头信息
     * @return 客户端的 IP 地址如果无法确定，则返回 "127.0.0.1"
     */
    public static String getIpAddress(HttpServletRequest request) {
        // 尝试从 'x-forwarded-for' 请求头获取 IP 地址
        String ip = request.getHeader("x-forwarded-for");
        if (ip == null || ip.length() == 0 || "unknown".equalsIgnoreCase(ip)) {
            // 如果 'x-forwarded-for' 头不存在或为未知值，则尝试从 'Proxy-Client-IP' 请求头获取 IP 地址
            ip = request.getHeader("Proxy-Client-IP");
        }
        if (ip == null || ip.length() == 0 || "unknown".equalsIgnoreCase(ip)) {
            // 如果 'Proxy-Client-IP' 头也不存在或为未知值，则尝试从 'WL-Proxy-Client-IP' 请求头获取 IP 地址
            ip = request.getHeader("WL-Proxy-Client-IP");
        }
        if (ip == null || ip.length() == 0 || "unknown".equalsIgnoreCase(ip)) {
            // 如果所有请求头都不存在或为未知值，则使用请求的远程地址作为 IP 地址
            ip = request.getRemoteAddr();
            if (ip.equals("127.0.0.1")) {
                // 如果远程地址是回环地址，则根据网卡取本机配置的 IP
                InetAddress inet = null;
                try {
                    inet = InetAddress.getLocalHost();
                } catch (Exception e) {
                    // 如果获取本机地址时发生异常，则记录异常信息
                    e.printStackTrace();
                }
                if (inet != null) {
                    ip = inet.getHostAddress();
                }
            }
        }
        // 处理多个代理的情况，取第一个 IP 作为客户端真实 IP
        if (ip != null && ip.length() > 15) {
            if (ip.indexOf(",") > 0) {
                ip = ip.substring(0, ip.indexOf(","));
            }
        }
        // 如果 IP 仍然为空，则默认返回回环地址
        if (ip == null) {
            return "127.0.0.1";
        }
        return ip;
    }

}
